package com.aurora.leetcode.answer;

import com.aurora.leetcode.util.MyUtils;

import java.util.HashMap;
import java.util.Map;

/**
 * @author : Aurora
 * @date : 2024/5/13 15:38
 * @description
 * <p>1. 两数之和</p>
 * <p>给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
 * 你可以按任意顺序返回答案。</p>
 * <p>exp1</p>
 * <p>输入：nums = [2,7,11,15], target = 9</p>
 * <p>输出：[0,1]</p>
 * <p>解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。</p>
 */
public class TwoSum_1 {

    //方法一：暴力枚举
    public int[] twoSum(int[] nums, int target) {
        int[] result = new int[2];
        for(int i=0; i<nums.length; ++i){
            int one = nums[i];
            for(int j=i+1; j<nums.length; ++j){
                int two = nums[j];
                if(target == one + two){
                    result[0] = i;
                    result[1] = j;
                    return result;
                }
            }
        }
        return result;
    }

    //方法二：哈希表
    public int[] twoSum2(int[] nums, int target) {
        Map<Integer, Integer> hashTable = new HashMap<>();
        for(int i=0; i<nums.length; ++i){
            if(hashTable.containsKey(target-nums[i])){
                return new int[]{i,hashTable.get(target-nums[i])};
            }
            hashTable.put(nums[i],i);
        }
        return new int[0];
    }

    public static void main(String[] args) {
//        int[] arr = new int[]{2,7,11,15};
//        int target = 9;
        int[] arr = new int[]{3,2,4};
        int target = 6;
        TwoSum_1 obj = new TwoSum_1();
//        int[] result = obj.twoSum(arr, target);
        int[] result = obj.twoSum2(arr, target);
        MyUtils.printArray(result);
    }
}
